package com.futao.controller;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;

/**
 * @author futao
 * Created on 18-6-11-下午4:02.
 * {@see https://www.cnblogs.com/xdp-gacl/p/4200090.html}
 */
@RestController
@Slf4j
public class FileUploadController {

    @RequestMapping("/fileUpload")
    public String fileUpload(
            HttpServletRequest httpServletRequest,
            HttpServletResponse httpServletResponse
    ) {
//        String savePath = httpServletRequest.getServletContext().getRealPath("/uploadFiles");
        String savePath = "./uploadFiles";
        File file = new File(savePath);
        if (!file.exists() && !file.isDirectory()) {
            log.warn(savePath + "目录不存在，需要创建!");
            file.mkdir();
        }
        String message = "未知异常";
        try {
            //使用Apache文件上传组件处理文件上传步骤：
            //1、创建一个DiskFileItemFactory工厂
            DiskFileItemFactory factory = new DiskFileItemFactory();
            //2、创建一个文件上传解析器
            ServletFileUpload upload = new ServletFileUpload(factory);
            //解决上传文件名的中文乱码
            upload.setHeaderEncoding("UTF-8");
            //4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
            List<FileItem> fileItems = upload.parseRequest(httpServletRequest);
            for (FileItem fileItem : fileItems) {
                //如果fileItem中封装的是普通输入项的数据
                if (fileItem.isFormField()) {
                    String fieldName = fileItem.getFieldName();
                    //解决普通输入项的数据的中文乱码问题
                    String value = fileItem.getString("UTF-8");
                } else {//如果fileitem中封装的是上传文件
                    //得到上传的文件名称
                    String fileName = fileItem.getName();
                    log.info("uploadFileName：" + fileName);
                    if (fileName == null || fileName.trim().equals("")) {
                        continue;
                    }
                    //注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如：  c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
                    //处理获取到的上传文件的文件名的路径部分，只保留文件名部分
                    fileName = fileName.substring(fileName.lastIndexOf("\\") + 1);
                    //获取item中的上传文件的输入流
                    InputStream fileItemInputStream = fileItem.getInputStream();
                    //创建一个文件输出流
                    FileOutputStream fileOutputStream = new FileOutputStream(savePath + "\\" + fileName);
                    //创建一个缓冲区
                    byte[] buffer = new byte[1024];
                    //判断输入流中的数据是否已经读完的标识
                    int len = 0;
                    //循环将输入流读入到缓冲区当中，(len=in.read(buffer))>0就表示in里面还有数据
                    while ((len = fileItemInputStream.read(buffer)) > 0) {
                        //使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
                        fileOutputStream.write(buffer, 0, len);
                    }
                    //关闭输入流
                    fileItemInputStream.close();
                    //关闭输出流
                    fileOutputStream.close();
                    //删除处理文件上传时生成的临时文件
                    fileItem.delete();
                    message = "文件上传成功！";
                }
            }
        } catch (FileUploadException | IOException e) {
            log.error(e.getMessage());
            e.printStackTrace();
            message = "文件上传失败";
        }
        log.info(this.getClass() + ": 上传文件结果 :" + message);
        return message;
    }
}
